package gulimall.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * @author HLF
 * @date 2022/8/20 17:02
 */
@Slf4j
@Configuration
public class RabbitMQConfig {

    @Resource
    private RabbitTemplate rabbitTemplate;

    /**
     * 配置消息的序列化为json格式
     *
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * <pre>
     * 定制RabbitTemplate
     * 1、服务器收到消息就回调
     *      1、spring.rabbitmq.publisher-confirm-type=correlated
     *      2、设置确认回调ConfirmCallback
     * 2、消息正确抵达队列进行回调
     *      1、 spring.rabbitmq.publisher-returns=true
     *          spring.rabbitmq.template.mandatory=true
     *      2、设置确认回调ReturnCallback
     * <p>
     * 3、消费端确认（保证每个消息被正确消费，此时才可以broker删除这个消息）。
     *      spring.rabbitmq.listener.simple.acknowledge-mode=manual 手动签收
     *      1、默认是自动确认的，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     *      问题：
     *          我们收到很多消息，自动回复给服务器ack，只有一个消息处理成功，宕机了。就会发生消息丢失；
     *          消费者手动确认模式。只要我们没有明确告诉MQ，货物被签收。没有Ack，
     *          消息就一直是unacked状态。即使Consumer宕机。消息不会丢失，会重新变为Ready，下一次有新的Consumer连接进来就发给他
     *      2、如何去签收:
     *      channel.basicAck(deliveryTag,false);签收；业务成功完成就应该签收
     *      channel.basicNack(deliveryTag,false,true);拒签；业务失败就拒签收
     * </pre>
     */
    @PostConstruct
    public void rabbitInit() {
        //ConfirmCallback为发送Exchange（交换器）时回调，成功或者失败都会触发；
        //ReturnCallback为路由不到队列时触发，成功则不触发

        /**
         * 设置消息发送方确认成功的回调
         * @param correlationData 消息相关数据
         * @param ack             broker是否接收到消息
         * @param cause           如果没有接收到消息，返回没有接收到的原因，反之返回null
         */
        rabbitTemplate.setConfirmCallback((correlationData, ack, cause) -> {
            String id = correlationData == null ? "" : correlationData.getId();
            if (ack) {
                log.info("消息发送方确认发送成功，消息被broker接收成功！消息的id是：{}", id);
            } else {
                log.info("消息发送方确认发送成功，消息未被broker接收，消息的id是：{}，原因是：{}", id, cause);
            }
        });

        /**
         * 设置消息确认失败的回调，被交换机接收，但是未被队列接收
         * @param message 哪个消息投递失败了，消息的具体内容
         * @param replyCode 回复的状态码
         * @param replyText 回复的文本内容
         * @param exchange 当前这个消息发给哪个交换机
         * @param routingKey 当前这个消息发送的时候，指定的哪个路由键
         */
        rabbitTemplate.setReturnsCallback(returned -> {
            Message message = returned.getMessage();
            int replyCode = returned.getReplyCode();
            String replyText = returned.getReplyText();
            String exchange = returned.getExchange();
            String routingKey = returned.getRoutingKey();

            log.info("消息发送方确认失败，消息具体内容：{}，回复状态码：{}，" +
                            "回复的文本内容：{}，当前这个消息发送给哪个交换机：{}，" +
                            "当前消息发送的时候，指定的routingKey：{}",
                    message, replyCode, replyText, exchange, routingKey);
        });
    }
}
